from pathlib import Path
import yaml

NEGATIVE_INF = -100000.0
HALF_NEGATIVE_INF = -60000.0  # half precision

# Config
CONFIG_FILE = Path('config.yml')

# Perspective API
# PERSPECTIVE_API_ATTRIBUTES = (
#     'TOXICITY',
#     'SEVERE_TOXICITY',
#     'IDENTITY_ATTACK',
#     'INSULT',
#     'THREAT',
#     'PROFANITY',
#     'SEXUALLY_EXPLICIT',
#     'FLIRTATION'
# )

# reward
# GAIN = 3.6595613471995
# BIAS = -2.6067927686264682
# GAIN = 34.1689259645464
# BIAS = -7.157455683003128

# maps clip score 0.5 to 0, 0.45 to -1
GAIN = 50
BIAS = -10
# STYLE_GAIN = 0.30046857952563216
# STYLE_BIAS = 1.8606487043321218
STYLE_ACC_GAIN = 2.8249959672299974
STYLE_ACC_BIAS = -0.4149256858992471
STYLE_PROB_GAIN = 4.5863538389169545
STYLE_PROB_BIAS = -0.49298448889514296
STYLE_GAIN = STYLE_PROB_GAIN
STYLE_BIAS = STYLE_PROB_BIAS
